﻿using System;
using System.Text;
using System.Collections.Generic;
using System.Linq;
using Microsoft.VisualStudio.TestTools.UnitTesting;
using System.Data;
using Task1;

namespace Main.Test
{
    [TestClass]
    public class MainTest
    {
        [TestMethod]
        public void FindMarginTest()
        {
            var dSet = new MyDataSet();

            dSet.Product.AddProductRow(152, "aaa", 10);
            dSet.Product.AddProductRow(152, "bbb", 5);
            dSet.Product.AddProductRow(152, "ccc", 10);
            dSet.Product.AddProductRow(152, "eee", 20);
            dSet.Product.AddProductRow(152, "fff", 20);

            dSet.Product.AddProductRow(146, "ccc", 15);
            dSet.Product.AddProductRow(146, "eee", 11);

            dSet.Product.AddProductRow(146, "aaa", 10);
            dSet.Product.AddProductRow(146, "bbb", 5);
            dSet.Product.AddProductRow(146, "ddd", 11);

            List<int> margins = WorkWithDataSet.FindMargin(dSet);

            Assert.AreEqual(2, margins.Count);
            Assert.AreEqual(new List<int> { 5, 9 }, margins);

        }
    }
}
